package william.list;

/**
 * @author ZhangShenao
 * @date 2024/3/9
 * @description <a href="https://leetcode.cn/problems/partition-list/description/">...</a>
 */
public class Leetcode86_分隔链表 {
    private class ListNode {
        int val;
        ListNode next;

        ListNode() {
        }

        ListNode(int val) {
            this.val = val;
        }

        ListNode(int val, ListNode next) {
            this.val = val;
            this.next = next;
        }
    }

    /**
     * 借助哑头节点实现
     * 维护两个哑头节点lt和ge,分别表示小于部分的链表头结点和大于等于部分链表的头节点
     * 遍历链表,将每个节点的值与基准值进行比较
     * - 如果小于基准值,则将节点追加到lt链表末尾
     * - 否则大于等于基准值,将节点追加到ge链表末尾
     * 最后将ge链表追加到lt链表末尾,返回lt链表的头节点
     * <p>
     * 时间复杂度O(N) 需要遍历一次链表
     * 空间复杂度O(1) 只额外申请了两个哑头节点
     */
    public ListNode partition(ListNode head, int x) {
        //边界条件校验
        if (head == null) {
            return head;
        }

        //维护两个哑头节点lt和ge,分别表示小于部分的链表头结点和大于等于部分链表的头节点
        ListNode ltHead = new ListNode();
        ListNode lt = ltHead;
        ListNode geHead = new ListNode();
        ListNode ge = geHead;
        ListNode cur = head;

        while (cur != null) {
            if (cur.val < x) {   //当前节点小于基准值,将节点追加到lt链表末尾
                lt.next = cur;
                lt = lt.next;
            } else {
                ge.next = cur;
                ge = ge.next;
            }
            cur = cur.next;
        }

        ge.next = null; //清空ge后的节点,避免成环

        //最后将ge链表追加到lt链表末尾,返回lt链表的头节点
        lt.next = geHead.next;
        return ltHead.next;
    }
}
